|
Server : LiteSpeed System : Linux host 5.15.0-161-generic #171-Ubuntu SMP Sat Oct 11 08:17:01 UTC 2025 x86_64 User : idnco5810 ( 1093) PHP Version : 8.2.29 Disable Function : NONE Directory : /usr/lib/python3/dist-packages/cloudinit/sources/__pycache__/ |
Upload File : |
o
쑛h�C � @ s� d Z ddlZddlZddlZddlmZ ddlmZmZmZ ddl Z ddl
m Z m Z m
Z
ddlmZ ddlmZ ddlmZ e�e�Zd Zd
Zd Zd Zd
ZdZ dZ de fdd�Z dee fdd�Z defdd�Z!defdd�Z"dedee fdd�Z#dee fd d �Z$G d d � d �Z%d edee& fd!d"�Z'dede de&fd#d$�Z(ded%e deee% ee% f fd&d'�Z)d(ee% d)ee% dee fd*d+�Z*d,e fd-d.�Z+G d/d0� d0e j,�Z-e-e j.ffgZ/d1d2� Z0dS )3z?Datasource to support the Windows Subsystem for Linux platform.� N)�PurePath)�List�Optional�Tuple)�sources�subp�util)�Distro)�type_from_starts_with)�Pathsz/usr/bin/wslpathziid-datasource-wsl� %s.user-dataz
agent.yaml� landscape�client�installation_request_id�returnc C s" t � tddg�\} }t| �� �jS )zL
Returns the name of the current WSL instance as seen from outside.
z-am�/)r � WSLPATH_CMDr �rstrip�name)�
root_net_path�_� r �A/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceWSL.py�
instance_name! s r c C sH d} d}g }t �� �� D ]}|d | kr!||d v r!|�|d � q |S )z�
Return a list of mount points of the Windows drives inside the current
WSL instance, if drives are mounted, or an empty list otherwise
�9pz aname=drvfs�fstype�opts�
mountpoint)r �mounts�values�append)�FS_TYPE�OPTIONS_CONTAIN�mounted�mntr r r �mounted_win_drives. s �r% c C s` t � } | s td��d}| D ]}|| }t�|tj�sq
t�d|� t|� S tdd�| � ��)z?
Returns the Linux path to the Windows host's cmd.exe.
z Windows drives are not mounted.z%s/Windows/System32/cmd.exezFound cmd.exe at <%s>z,Couldn't find cmd.exe in any mount point: %sz, ) r% �IOError�os�access�X_OK�LOG�debugr �join)r � candidater$ �cmdr r r �cmd_executable> s �r/ c C sV t � } t�d| �� ddg�\}}|�� }|st�d��t�td|g�\}}t|�� �S )z�
Finds the user's home directory path as a WSL path.
raises: IOError when no mountpoint with cmd.exe is found
ProcessExecutionError when either cmd.exe is unable to retrieve
the user's home directory
z/initz/Czecho %USERPROFILE%z4No output from cmd.exe to show the user profile dir.z-au)r/ r �as_posixr �ProcessExecutionErrorr r )r. �homer �outr r r � find_homeV s � r4 � user_homec C s2 t j�| d�}t j�|�st�d|� dS t|�S )z�
Returns the Windows user profile .cloud-init directory translated as a
Linux path accessible inside the current WSL instance, or None if not
found.
z .cloud-initz*cloud-init user data dir %s doesn't exist.N)r' �pathr, �isdirr* r+ r )r5 �seed_dirr r r �cloud_init_data_dirt s
r9 c C s6 t �� \}}}|r |n|}d| d||f d| dgS )z
Return a list of candidate file names that may contain user-data
in some supported format, ordered by precedence.
r z%s-%s.user-dataz%s-all.user-datazdefault.user-data)r �get_linux_distro)r �distribution_id�
version_id�version_codename�versionr r r � candidate_user_data_file_names� s
�r? c @ s0 e Zd ZdZdefdd�Zdedefdd�Zd S )
�
ConfigDataz~Models a piece of configuration data as a dict if possible, while
retaining its raw representation alongside its file pathr c C s
| j d uS �N)� config_dict)�selfr r r �is_cloud_config� s
zConfigData.is_cloud_configr6 � instance_idc C s� t �|�| _|| _d | _dt| j�krt �| j�| _| jrK|rM|tkrO| j�t i ��t
�d urQ| jt t
� t |�}||krSdt
�| j��� | _d S d S d S d S d S d S )Nztext/cloud-configz#cloud-config
%s)r �load_text_file�rawr6 rB r
� load_yaml�DEFAULT_INSTANCE_ID�get�LANDSCAPE_CFG_KEY�LANDSCAPE_CLIENT_CFG_KEY�
setdefault� LANDSCAPE_INSTALLATION_REQ_ID�yaml�dump�strip)rC r6 rE �req_idr r r �__init__� s6 ��������
zConfigData.__init__N) �__name__�
__module__� __qualname__�__doc__�boolrD r �strrS r r r r r@ � s r@ �
metadata_pathc C s@ d}z
t jt �| �i d�}W |S ty t�d| � Y |S w )zO
Returns the relevant metadata loaded from the supplied metadata_path.
N)�defaultz!No instance metadata found at %s.)r rH rF �FileNotFoundErrorr* r+ )rZ �metadatar r r �_load_metadata� s
� ���r^ c
C s� dt i}t| d �}t| �}d}||fD ] }|du rq|d| }t|�}|dur1|}|�� } nq|s9t�d� |r?d|vrPd|� d|� �} t�| � t| ��|S ) z�
Returns the relevant metadata loaded from either the Ubuntu Pro dir
or the user cloud-int dir based on the instance name
� instance-id�
.ubuntupro� Nz %s.meta-datazCUnable to find meta-data file candidates. Using default instance-idz Metadata at z5 does not contain instance-id key. Instead received: )rI r9 r^ r0 r* r+ �error�
ValueError)
r5 r r] �pro_dir�user_dir�found_at�dirr6 �dt�msgr r r �load_instance_metadata� s4 �� ��
rj rE c C s� t | d �}|du r dS ttj�|tt� ��}d}tj�|�r.t� d|t | �� t
||�}ttj�|t ��}d}tj�|�rDt
||�}||fS )zg
Read .ubuntupro user-data if present and return a tuple of agent and
landscape user-data.
r` N)NNzYLandscape configuration found: %s. Organization policy ignores any local user-data in %s.) r9 r r' r6 r, �LANDSCAPE_DATA_FILEr �isfiler* r+ r@ �AGENT_DATA_FILE)r5 rE rd �landscape_path�landscape_data�
agent_path�
agent_datar r r �load_ubuntu_pro_data� s& � �
rr rq � user_datac C s\ | du s| j s|du s|j sdS |j S |du s|j s&| du s!| j s#dS | j S t| �� |�� g�sAt�d� d| j�� |j�� f S i }d}g }t|jt �r^|j}|�
di ��
di ��
dd�}t| jt �r�|rkt�d� | j}|D ]}||v r{|� |� || ||< qp|r�t�d � |�� |r�|�
di ��
d�r�t�d
|� ||d d d< d t
�|��� S ) a Merge agent.yaml data provided by Ubuntu Pro for WSL
and user data provided either by Landscape or the local user,
according to the UP4W specific rules.
When merging is not possible, provide #include directive to allow
cloud-init to merge separate parts.
NzaUnable to merge {agent_data.path} and {user_data.path}. Providing as separate user-data #include.z#include
%s
%s
ra r
r �tagsz.Merging both user_data and agent.yaml configs.z% agent.yaml overrides config keys: , zFLandscape client conf updated with user-data landscape.client.tags: %sz@#cloud-config
# WSL datasouce Merged agent.yaml and user_data
%s)rG �allrD r* r+ r6 r0 �
isinstancerB �dictrJ r r, rO rP rQ )rq rs �merged� user_tags�overridden_keys�agent�keyr r r �merge_agent_landscape_data s\ �� �
�� ��r} �fieldc
C sf zd| � dd� }t�ddg�\}}||v W S tjy2 } zt�d| |j� W Y d}~dS d}~ww ) z�Checks if the landscape-config binary supports the intended
configuration field, returning False if not or if attempting
to run that binary fails.
z--r �-zlandscape-configz--helpz4Unable to verify if landscape-client supports %s: %sNF)�replacer r1 r* �warning�reason)r~ �flagr3 r �errr r r �landscape_supports_fieldf s
���r� c sZ e Zd ZdZddedef� fdd�
Zdedefd d
�Zde fd d �Z
de fd
d�Z � Z S )�
DataSourceWSL�WSLN�distro�pathsc s t � �||||� d| _d S )Nra )�superrS r )rC �sys_cfgr� r� �ud_proc�� __class__r r rS ~ s
zDataSourceWSL.__init__r8 r c C sj dd� t �|�D �}|std| ��dd� t| j�D �}|D ]}||�� v r.t|| � S q td| ��)z�
Finds the most precendent of the candidate files that may contain
user-data, if any, or None otherwise.
c S s i | ] }|j �� |j�qS r )r �casefoldr6 )�.0�efr r r �
<dictcomp>� s �z5DataSourceWSL.find_user_data_file.<locals>.<dictcomp>z%s directory is emptyc S s g | ]}|� � �qS r )r� )r� �fr r r �
<listcomp>� s ��z5DataSourceWSL.find_user_data_file.<locals>.<listcomp>z6%s doesn't contain any of the expected user-data files)r' �scandirr& r? r �keysr )rC r8 �existing_files� folded_names�filenamer r r �find_user_data_file� s � � ��z!DataSourceWSL.find_user_data_filec
C sn | � � }|sdS zt� }t|t� �}||�d�kW S ttfy6 } zt�dt |�� W Y d }~dS d }~ww )NFr_ z2Unable to check_instance_id from metadata file: %s)
�get_instance_idr4 rj r rJ r& rc r* r� rY )rC r� �currentr2 r] r� r r r �check_instance_id� s ���z DataSourceWSL.check_instance_idc
C s| t �t�s
t�dt� dS t� | _zt� }W n ty/ } z
t�d|� W Y d }~dS d }~ww t|�}d }d }z t || j�| _
W n t tfy^ } zt� dt
|�� W Y d }~dS d }~ww | j
d }| jjdkrwtt�spd}t||�\}}z|d u r�|d ur�t| �|�d�}W n$ t tfy� } z|r�tjntj }|d|t
|�� W Y d }~nd }~ww t||g�s�dS t||�| _d S )
Nz5No WSL command %s found. Cannot detect WSL datasourceFz#Unable to detect WSL datasource: %szUnable to load metadata: %sr_ �ubuntura z+Unable to load any user-data file in %s: %sT)r �whichr r* r+ r r4 r& r9 rj r] rc rb rY r� r r� rN rr r@ r� �info�anyr} � userdata_raw) rC r5 �er8 rq rs r� �iid�logr r r � _get_data� s`
�
��
���
���� zDataSourceWSL._get_datarA )
rT rU rV �dsnamer r rS r r� rX r� r� �
__classcell__r r r� r r� { s r� c C s t �| t�S rA )r �list_from_depends� datasources)�dependsr r r �get_datasource_list� s r� )1rW �loggingr' �typing�pathlibr r r r rO � cloudinitr r r �cloudinit.distrosr �cloudinit.handlersr
�cloudinit.helpersr � getLoggerrT r* r rI rk rm rK rL rN rY r r% r/ r4 r9 r? r@ rw r^ rj rr r} r� �
DataSourcer� �DEP_FILESYSTEMr� r� r r r r �<module> s^
$%��
� ��
�U
u�